# plot_multiple_columns_from_dat_float.py
import matplotlib.pyplot as plt
import matplotlib
matplotlib.rc("font",family='YouYuan')
from datetime import datetime

# x = [157, 198, 259, 294, 328, 361, 389]
# y = [7.3, 11.5, 14, 16,  14.2,11.5, 10.1]

x = [140, 158, 170, 180, 200, 220, 240, 280]
y = [2.54, 3.27, 3.73, 4.1,4.0,4.1, 3.9, 2.3]


plt.figure()
# plt.grid(visible=True, which=u'major', axis=u'x') # 添加网格线
# plt.grid(visible=True, which=u'major') # 添加网格线

# 创建横坐标序列（现在序号*10）
# 绘制每个点的横纵两条直线
# for i in range(len(x)):
#     plt.plot([x[i], x[i]], [0, y[i]], 'r--', linewidth=1)  # 绘制纵向直线
#     plt.plot([0, x[i]], [y[i], y[i]], 'r--', linewidth=1)  # 绘制横向直线


plt.plot(x, y,  marker='o', label='sss')
# plt.scatter(x, y, s=100, marker='o')

# 在每个点的右上方显示坐标
for i in range(len(x)):
    plt.text(x[i] + 0.1, y[i] + 0.1, f'({x[i]}, {y[i]})', fontsize=14, ha='left')


plt.xlabel('伸腿长度 ( mm )')
plt.ylabel('电机力矩 ( n.m )')
# 保存图片为.png格式，文件名为当前日期和时间（精确到秒）
current_datetime = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
plt.savefig(f"plot_{current_datetime}.png", dpi=300)
plt.show()
